# ---- To start a beefed up interactive session ----# 
module load gpu 
module load cuda/11.8.0
srun --pty -n 1 -c 16 --time=02:00:00 --gpus=1 --constraint="GPUMEM80GB"  --mem=32G bash -l
nvidia-smi   # to check the GPU you have been allocated
module load anaconda3
module load mamba # (if you need it)
conda activate LLM_v5
python3 LLM_static.py   # now you can see the gpus!


# ---- To submit a job to the cluster ----#
module load a100 # since you only want a100 GPUs 
module load cuda/11.8.0
module load anaconda3
sbatch cluster_script.sh
```